import json
import pygal
from pygal.style import LightColorizedStyle as LCS, RotateStyle as RS

import country_func as fun

"""世界人口地图"""

# 将数据加载到一个列表中
filename = 'population_data.json'
with open(filename) as f:
    pop_data = json.load(f)

# 创建一个包含人口数量的字典
# cc_populations = {}
# 根据人口数量将所有的国家分成三组
cc_pops_1, cc_pops_2, cc_pops_3 = {}, {}, {}
for pop_dict in pop_data:
    if pop_dict['Year'] == 2010:
        country_name = pop_dict['Country Name']
        population = pop_dict['Value']  # 人口数量
        country_code = fun.get_country_code(country_name)  # 国别代码
        if population < 10000000:
            cc_pops_1[country_code] = population
        elif population < 100000000:
            cc_pops_2[country_code] = population
        else:
            cc_pops_3[country_code] = population
# 看看每组分别包含多少个国家
print(len(cc_pops_1), len(cc_pops_2), len(cc_pops_3))
# 颜色设置
# wm_style = RotateStyle('#005500')
# wm_style = LightColorizedStyle
wm_style = RS('#336699', base_style=LCS)

wm = pygal.maps.world.World(style=wm_style)

wm.title = 'World Population in 2010, by Country'
wm.add('0-10m', cc_pops_1)
wm.add('10m-1bn', cc_pops_2)
wm.add('>1bn', cc_pops_3)
wm.render_to_file('world_population.svg')

